"""
练习案例：读取文件中的数据，并将数据插入到数据库
"""
from pymysql import Connection

from my_package.file_define import TextFileReader, JsonFileReader, read_file
from my_package.data_define import ReCord

text_file_path = "/Users/dongjian/Documents/学习/python-learn-资料/资料/第13章资料/2011年1月销售数据.txt"
json_file_path = "/Users/dongjian/Documents/学习/python-learn-资料/资料/第13章资料/2011年2月销售数据JSON.txt"

text_file_reader = TextFileReader(text_file_path)
json_file_reader = JsonFileReader(json_file_path)

text_list1 = read_file(text_file_reader)
json_list2 = read_file(json_file_reader)

# 合并数据
all_data: list[ReCord] = text_list1 + json_list2

conn = Connection(
    host="localhost",
    port=3306,
    user="root",
    password="root",
    database="py_sql",
    autocommit=True
)
# 获取游标对象
cursor = conn.cursor()
for data in all_data:
    sql = f"insert into orders(order_date, order_id, money, province)" \
          f" values('{data.date}', '{data.order_id}', '{data.money}', '{data.province}')"
    # 执行sql语句
    cursor.execute(sql)
    # print(sql)
# 关闭连接
conn.close()
